import pandas as pd
import matplotlib.pyplot as plt

# the training losses are saved in the model_dir
df = pd.read_csv("model/loss.csv")
df[["train_loss", "val_loss"]].ffill().plot(grid=True)
plt.show()
